package com.itany.shop.util;

import com.itany.shop.exception.DataAccessException;

import java.sql.*;
import java.util.ArrayList;
import java.util.List;

/**
 * @author MissChen
 * @version 1.0
 * @date 2023 年12月20日 13:58
 */
public class JDBCTemplate<T> {

    /**
     * 一个DML操作，insert、update、delete
     * 对于增删改操作,ps.executeUpdate()
     * 对于整个方法而言，sql语句不一样，以及参数不一样，其他代码基本相同
     * @param sql
     * @param params
     */
    public void update(String sql, Object... params) {
        Connection conn = null;
        PreparedStatement ps = null;
        try {
            conn = JDBCUtil.getConnection();
            ps=conn.prepareStatement(sql);

            //对参数进行操作
            for (int i = 0; i < params.length; i++) {
                Object param = params[i];
                if (param instanceof String) {
                    ps.setString(i + 1, (String) param);
                } else if (param instanceof Integer) {
                    ps.setInt(i + 1, (Integer) param);
                } else if (param instanceof Double) {
                    ps.setDouble(i + 1, (Double) param);
                } else {
                    ps.setObject(i + 1, param);
                }

            }
            //执行DML操作
            ps.executeUpdate();

        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("数据库访问异常");
        }finally {
            //在业务层的事务操作中需要使用Connection
            // 因此，此时释放资源时候不要释放Connection
            JDBCUtil.close(null,ps,null);
        }
    }

    /**
     * 执行查询操作
     * 查询会返回一个结果，有可能是一条，也有可能是多条
     * @param sql  执行的sql语句
     * @param rm 行映射，将查询结果映射成Java对象
     * @param params 执行sql所需要的参数
     * @return
     */
    public List<T> query(String sql,RowMapper rm,Object... params){
        Connection conn = null;
        PreparedStatement ps = null;
        ResultSet rs=null;
        List<T> list=new ArrayList<>();
        try {
            conn=JDBCUtil.getConnection();
            ps=conn.prepareStatement(sql);
            //对参数进行操作
            for (int i = 0; i < params.length; i++) {
                Object param = params[i];
                if (param instanceof String) {
                    ps.setString(i + 1, (String) param);
                } else if (param instanceof Integer) {
                    ps.setInt(i + 1, (Integer) param);
                } else if (param instanceof Double) {
                    ps.setDouble(i + 1, (Double) param);
                } else {
                    ps.setObject(i + 1, param);
                }

            }
            rs=ps.executeQuery();

            //执行查询操作
            while(rs.next()){
                T t= (T) rm.mapRow(rs);
                list.add(t);
            }
            return list;

        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("数据库访问异常");
        }finally {
            JDBCUtil.close(null,ps,rs);
        }
    }



    public void insertReturnPrimaryKey(PreparedStatementCreator psc,KeyHolder keyHolder){
        Connection conn=null;
        PreparedStatement ps=null;
        ResultSet rs=null;
        try {
            conn=JDBCUtil.getConnection();

            //调用者在调用方法的时候做具体的实现
            // 在实现的时候，会提供构建状态集时所需要的sql与参数
            ps=psc.createPreparedStatement(conn);
            ps.executeUpdate();

            //执行完添加后会自动生成主键
            // 获取生成的主键
            rs=ps.getGeneratedKeys();

            //准备一个集合，用于存储自动生成的主键的值
            List<Integer> list=new ArrayList<>();

            //通过结果集元数据
            ResultSetMetaData rsmd=rs.getMetaData();

            //获取生成的列的总数
            int columnCount=rsmd.getColumnCount();
            System.out.println(columnCount+"-------------------------");
            if(rs.next()){
                for(int i=1;i<=columnCount;i++){
                list.add(rs.getInt(i));
            }
            keyHolder.setKeyList(list);
        }

        } catch (SQLException e) {
            e.printStackTrace();
            throw new DataAccessException("数据库访问异常");
        }finally {
            JDBCUtil.close(null,ps,rs);
        }

    }

}
